TRAINING_PARAMS = \
{
    "model_params": {
        "backbone_name": "darknet_53",
        "backbone_pretrained": "../weights/darknet53_weights_pytorch.pth",  # set empty to disable
    },
    "yolo": {
        "anchors": [[[152, 98], [201, 149], [210, 50]],
                    [[67, 222], [78, 150], [134, 214]],
                    [[40, 90], [52, 121], [62, 126]]],
        "classes": 80,
    },
    "lr": {
        "backbone_lr": 0.001,
        "other_lr": 0.01,
        "freeze_backbone": False,   # freeze backbone wegiths to finetune
        "decay_gamma": 0.1,
        "decay_step": 20,           # decay lr in every ? epochs
    },
    "optimizer": {
        "type": "sgd",
        "weight_decay": 4e-05,
    },
    "batch_size": 16,
    "train_path": "/mnt/sda1/hgl/indicator/indicator_image_416_sorted.txt",
    "train_label_path": "/mnt/sda1/hgl/indicator/label_normalize/indicator_indicator_sorted.txt",
    "epochs": 400,
    "img_h": 416,
    "img_w": 416,
    "parallels": [2],                             # config GPU device
    "working_dir": "../bobliu_yolov3",      # replace with your working dir
    "pretrain_snapshot": "",    # load checkpoint
    "evaluate_type": "",
    "try": 5,
    "export_onnx": False,
}
